package cn.edu.nwpu.warehousemanagement.dao;

import cn.edu.nwpu.warehousemanagement.pojo.Product;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Modifying;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;
import org.springframework.transaction.annotation.Transactional;

import java.util.List;
import java.util.Optional;
import java.util.Set;
import java.util.stream.Stream;

@Repository
public interface ProductRepository extends JpaRepository<Product, Long> {

    // ==================== 基础CRUD ====================
    // 继承自JpaRepository:
    // save()       - 创建/更新产品
    // findById()   - 根据ID查询（默认不包含已删除的）
    // deleteById() - 改为软删除
    // findAll()    - 查询所有未删除的产品

    // ==================== 自定义查询 ====================
    //精确匹配ID
    Optional<Product> findByIdAndDeletedFalse(Long id);

    // 精确匹配产品名称（仅查询未删除的）
    Optional<Product> findByNameAndDeletedFalse(String name);

    // 名称模糊查询（区分大小写，仅查询未删除的）
    List<Product> findByNameContainingAndDeletedFalse(String keyword);

    // 名称模糊查询（不区分大小写，仅查询未删除的）
    List<Product> findByNameContainingIgnoreCaseAndDeletedFalse(String keyword);

    // 根据单位查询（仅查询未删除的）
    List<Product> findByUnitAndDeletedFalse(String unit);

    // 查询库存量大于指定值的产品（仅查询未删除的）
    List<Product> findByQuantityGreaterThanAndDeletedFalse(Long quantity);

    // 查询库存量小于指定值的产品（仅查询未删除的）
    List<Product> findByQuantityLessThanAndDeletedFalse(Long quantity);

    // 查询库存量在区间内的产品（仅查询未删除的）
    List<Product> findByQuantityBetweenAndDeletedFalse(Long min, Long max);

    // 在ProductRepository.java中添加以下方法：

    /**
     * 根据ID列表查询多个未删除的产品
     * @param ids 产品ID列表
     * @return 匹配的产品列表
     */
    List<Product> findByIdInAndDeletedFalse(List<Long> ids);

    // 根据描述模糊查询（仅查询未删除的）
    @Query("SELECT p FROM Product p WHERE p.description LIKE %:keyword% AND p.deleted = false")
    List<Product> searchByDescription(@Param("keyword") String keyword);

    // ==================== 更新操作 ====================
    // 更新产品名称（仅更新未删除的）
    @Modifying
    @Transactional
    @Query("UPDATE Product p SET p.name = :name WHERE p.id = :id AND p.deleted = false")
    int updateNameById(@Param("id") Long id, @Param("name") String name);

    // 更新库存量（增量更新，仅更新未删除的）
    @Modifying
    @Transactional
    @Query("UPDATE Product p SET p.quantity = p.quantity + :delta WHERE p.id = :id AND p.deleted = false")
    int updateQuantity(@Param("id") Long id, @Param("delta") Long delta);

    // 批量更新单位（仅更新未删除的）
    @Modifying
    @Transactional
    @Query("UPDATE Product p SET p.unit = :unit WHERE p.id IN :ids AND p.deleted = false")
    int batchUpdateUnit(@Param("ids") List<Long> ids, @Param("unit") String unit);

    // ==================== 删除操作 ====================
    // 软删除 - 根据ID删除
    @Modifying
    @Transactional
    @Query("UPDATE Product p SET p.deleted = true WHERE p.id = :id")
    void softDeleteById(@Param("id") Long id);

    // 软删除 - 根据名称删除
    @Modifying
    @Transactional
    @Query("UPDATE Product p SET p.deleted = true WHERE p.name = :name")
    void softDeleteByName(@Param("name") String name);

    // 批量软删除库存量为0的产品
    @Modifying
    @Transactional
    @Query("UPDATE Product p SET p.deleted = true WHERE p.quantity = :quantity")
    void softDeleteByQuantityEquals(@Param("quantity") Long quantity);

    // ==================== 统计查询 ====================
    // 检查名称是否存在（仅检查未删除的）
    boolean existsByNameAndDeletedFalse(String name);

    // 检查名称是否存在（排除指定ID的记录，仅检查未删除的）
    boolean existsByNameAndIdNotAndDeletedFalse(String name, Long id);

    // 统计指定单位的商品数量（仅统计未删除的）
    long countByUnitAndDeletedFalse(String unit);

    // 获取总库存量（仅统计未删除的）
    @Query("SELECT SUM(p.quantity) FROM Product p WHERE p.deleted = false")
    Long sumAllQuantities();

    // ==================== 分页查询 ====================
    // 分页查询未删除的产品
    Page<Product> findByDeletedFalse(Pageable pageable);

    // ==================== 复杂查询 ====================
    // 联合InventoryConfig查询低库存产品（仅查询未删除的）
    @Query("SELECT p FROM Product p JOIN InventoryConfig ic ON p.id = ic.product.id " +
            "WHERE p.quantity < ic.safetyStock AND p.deleted = false")
    List<Product> findLowStockProducts();

    // 分页查询（通常在Service层结合Pageable使用，仅查询未删除的）
    @Query("SELECT p FROM Product p WHERE p.deleted = false ORDER BY p.quantity DESC")
    List<Product> findAllOrderByQuantityDesc();

    // ==================== 包含已删除记录的查询 ====================
    // 查询所有记录（包括已删除的）
    @Query("SELECT p FROM Product p")
    List<Product> findAllIncludingDeleted();

    // 根据ID查询（包括已删除的）
    @Query("SELECT p FROM Product p WHERE p.id = :id")
    Optional<Product> findByIdIncludingDeleted(@Param("id") Long id);

    // 恢复已删除的记录
    @Modifying
    @Transactional
    @Query("UPDATE Product p SET p.deleted = false WHERE p.id = :id")
    void restoreById(@Param("id") Long id);

    // 根据ID集合查询未删除的商品
    List<Product> findByIdInAndDeletedFalse(Set<Long> ids);


    // 一次性获取所有产品统计信息（原生SQL优化性能）
    @Query(value =
            "SELECT " +
                    "  p.id AS product_id, " +
                    "  p.name AS product_name, " +
                    "  COUNT(CASE WHEN f.status = 'SHIPPED' THEN 1 END) AS shipped_count, " +
                    "  COUNT(CASE WHEN f.status = 'IN_STOCK' THEN 1 END) AS in_stock_count " +
                    "FROM product p " +
                    "LEFT JOIN freight f ON p.id = f.product_id AND f.deleted = false " +
                    "WHERE p.deleted = false " +
                    "GROUP BY p.id, p.name",
            nativeQuery = true)
    List<Object[]> findAllFreightStats();

    // 统计产品种类数量（未删除）
    @Query("SELECT COUNT(p) FROM Product p WHERE p.deleted = false")
    long countByDeletedFalse();

    // 统计已用货架数（有在库Freight的未删除货架）
    @Query("SELECT COUNT(DISTINCT f.shelf.id) " +
            "FROM Freight f " +
            "WHERE f.status = 'IN_STOCK' " +
            "AND f.deleted = false " +
            "AND f.shelf.deleted = false")
    long countUsedShelves();
}